Abstract:In data-intensive domains like medical diagnosis and financial risk control, feature selection is faced with severe challenges due to high-dimensional imbalanced data, including majority class bias, insufficient capture of high-order nonlinear feature interactions, and difficulty in collaborative optimization of conflicting multi-objectives. Traditional methods struggle to simultaneously achieve imbalanced data adaptation, high-order feature correlation mining, and multi-objective global optimization. Therefore, for supervised feature selection scenarios, a deep learning based multi-objective multi-task evolutionary approach for imbalanced feature selection(DLME) is proposed and a three-stage modular collaborative optimization framework is established. First, dynamic oversampling and preliminary feature filtering are performed by combining adaptive sampling and SMOTE based on particle swarm optimization, thereby alleviating class distribution imbalance. Then, a deep neural network is embedded. High-order nonlinear correlations among features are learned from the balanced sample set. The feature subset is subsequently refined. Finally, a grey wolf optimizer based multi-task multi-objective optimization framework is constructed. This framework integrates NSGA-II-based Pareto front search and front knowledge transfer mechanism. The optimal feature subset balancing classification accuracy and sparsity is output. Experimental results on public datasets and domain-specific datasets demonstrate the outstanding superiority of DLME under extremely imbalanced and high-dimensional scenarios. DLME exhibits excellent multi-objective optimization performance and robustness, thereby providing an effective solution for feature selection tasks on high-dimensional imbalanced data.
[1] Luo C, Wang S Z, Li T R, et al. Large-scale meta-heuristic feature selection based on BPSO assisted rough hypercuboid approach[J]. IEEE Transactions on Neural Networks and Learning Systems, 2023, 34(12): 10889-10903. [2] Mafarja M M, Mirjalili S.Hybrid whale optimization algorithm with simulated annealing for feature selection[J]. Neurocomputing, 2017, 260: 302-312. [3] Yang J Q, Yang Q T, Du K J, et al. Bi-directional feature fixation-based particle swarm optimization for large-scale feature selection[J]. IEEE Transactions on Big Data, 2023, 9(3): 1004-1017. [4] Zhang Y S, Wu J, Cai Z H, et al. Multi-view multi-label learning with sparse feature selection for image annotation[J]. IEEE Tran-sactions on Multimedia, 2020, 22(11): 2844-2857. [5] Jiao R W, Nguyen B H, Xue B, et al. A survey on evolutionary multiobjective feature selection in classification: approaches, applications, and challenges[J]. IEEE Transactions on Evolutionary Computation, 2024, 28(4): 1156-1176. [6] Ding W P, Lin C T, Pedrycz W.Multiple Relevant feature ensemble selection based on multilayer co-evolutionary consensus MapReduce[J]. IEEE Transactions on Cybernetics, 2020, 50(2): 425-439. [7] Chawla N V, Bowyer K W, Hall L O, et al. SMOTE: synthetic minority over-sampling technique[J]. Journal of Artificial Intelligence Research, 2002, 16: 321-357. [8] Maldonado S, Vairetti C, Fernandez A,et al. FW-SMOTE: a feature-weighted oversampling approach for imbalanced classification[J/OL]. Pattern Recognition, 2022, 124. https://doi.org/10.1016/j.patcog.2021.108511. [9] Bui L T, van Truong V U, Dinh T T H, et al. A multi-objective co-operative co-evolutionary method for classification with imbalanced data[C/OL]//Proceeding of the 15th International Conference on Knowledge and Systems Engineering. Washington, USA: IEEE, 2023. https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=10299473. [10] Liu J H, Lin Y J, Ding W P, et al. Fuzzy mutual information-based multilabel feature selection with label dependency and streaming labels[J]. IEEE Transactions on Fuzzy Systems, 2023, 31(1): 77-91. [11] Shu J, Yuan X, Meng D Y, et al. CMW-Net: learning a class-aware sample weighting mapping for robust deep learning[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2023, 45(10): 11521-11539. [12] Zhao H, Wang P, Hu Q H, et al. Fuzzy rough set based feature selection for large-scale hierarchical classification[J]. IEEE Transactions on Fuzzy Systems, 2019, 27(10): 1891-1903. [13] Mirjalili S, Mirjalili S M, Lewis A.Grey wolf optimizer[J]. Advances in Engineering Software, 2014, 69: 46-61. [14] Deb K, Pratap A, Agarwal S, et al. A fast and elitist multiobjective genetic algorithm: NSGA-II[J]. IEEE Transactions on Evolutionary Computation, 2002, 6(2):182-197. [15] Yuan Z, Chen H M, Zhang P F, et al. A novel unsupervised app-roach to heterogeneous feature selection based on fuzzy mutual information[J]. IEEE Transactions on Fuzzy Systems, 2022, 30(9): 3395-3409. [16] Zheng W, Chen S, Fu Z Y, et al. Feature selection boosted by unselected features[J]. IEEE Transactions on Neural Networks and Learning Systems, 2022, 33(9): 4562-4574. [17] Chakraborty R, Pal N R.Feature selection using a neural framework with controlled redundancy[J]. IEEE Transactions on Neural Networks and Learning Systems, 2015, 26(1): 35-50. [18] Zhou P, Li P P, Zhao S, et al. Feature interaction for streaming feature selection[J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 32(10): 4691-4702. [19] Shaham U, Lindenbaum O, Svirsky J, et al. Deep unsupervised feature selection by discarding nuisance and correlated features[J]. Neural Networks, 2022, 152: 34-43. [20] Gong X L, Yu L, Wang J, et al. Unsupervised feature selection via adaptive autoencoder with redundancy control[J]. Neural Networks, 2022, 150: 87-101. [21] Kononenko I.Estimating attributes: analysis and extensions of Relief[C]//Proceedings of the European Conference on Machine Learning. Berlin, Germany: Springer, 1994: 171-182. [22] PENG H C, LONG F H, DING C.Feature selection based on mutual information criteria of max-dependency, max-relevance, and min-redundancy[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2005, 27(8): 1226-1238. [23] Zhang Y H, Lu Z M, Wang S P. Unsupervised feature selection via transformed auto-encoder[J/OL]. Knowledge-Based Systems, 2021, 215. https://doi.org/10.1016/j.knosys.2021.106748. [24] Ling Y Z, Nie F P, Yu W Z, et al. Discriminative and robust auto-encoders for unsupervised feature selection[J]. IEEE Transactions on Neural Networks and Learning Systems, 2025, 36(1): 1622-1636. [25] Guyon I, Weston J, Barnhill S, et al. Gene selection for cancer classification using support vector machines[J]. Machine Lear-ning, 2002, 46(1): 389-422. [26] Alsalamah A H, Ismail N W. Evolutionary computation for feature optimization and image-based dimensionality reduction in IoT intrusion detection[J/OL]. Mathematics, 2025, 13(23). https://doi.org/10.3390/math13233869. [27] Tibshirani R.Regression shrinkage and selection via the Lasso[J]. Journal of the Royal Statistical Society(Methodological), 1996, 58(1): 267-288. [28] Xue B, Zhang M J, Browne W N, et al. A survey on evolutionary computation approaches to feature selection[J]. IEEE Transactions on Evolutionary Computation, 2016, 20(4): 606-626. [29] Ding W P, Zhou T Y, Huang J S, et al. FMDNN: a fuzzy-guided multigranular deep neural network for histopathological image cla-ssification. IEEE Transactions on Fuzzy Systems, 2024, 32(8): 4709-4723. [30] 华启冲,丁卫平,陈悦鹏,等.基于直觉模糊深度自动编码器的特征选择算法[J].计算机研究与发展,2026, 63(6): 1548-1562. (Hua Q C, Ding W P, Chen Y P, et al. Intuitionistic fuzzy deep autoencoder algorithm for feature selection[J]. Journal of Compu-ter Research and Development, 2026, 63(6): 1548-1562.) [31] Chen Y P, Ding W P, Gao S C,et al. Uncertainty-aware multi-view evidence fusion for feature selection in brain network analysis[J/OL]. Information Fusion, 2026, 130. https://doi.org/10.1016/j.inffus.2025.104083. [32] Liu Y, Li G S, Li X, ,et al. The classification method based on evo-lutionary algorithm for high-dimensional imbalanced missing data[J/OL]. Electronics Letters, 2023, 59(12). https://doi.org/10.1049/ell2.12842. [33] Amri A A, Ismail A R, Mohammad O A.Evolutionary deep belief networks with bootstrap sampling for imbalanced class datasets[J]. International Journal of Advanced Intelligence and Informatics, 2019, 5(2): 123-136. [34] Nasim A, Kim Y S. DE-PNN: differential evolution-based feature optimization with probabilistic neural network for imbalanced arrhythmia classification[J/OL]. Sensors, 2022, 22(12). https://doi.org/10.3390/s22124450. [35] WU X Y, WU S H, WU J B, et al. Evolutionary computation in the era of large language model: survey and roadmap. IEEE Tran-sactions on Evolutionary Computation, 2025, 29(2): 534-554. [36] Wang R, Bian J T, Nie F P, et al. Nonlinear feature selection neural network via structured sparse regularization[J]. IEEE Transactions on Neural Networks and Learning Systems, 2023, 34(11): 9493-9505. [37] Yuan X Y, Wei S L, Sun Y,et al. Robust multi-task feature selection with counterfactual explanation for schizophrenia identification using functional brain networks[J/OL]. Frontiers in Neuroscience, 2025, 19. https://doi.org/10.3389/fnins.2025.1609547. [38] Balin M F, Abid A, Zou J.Concrete autoencoders for differen-tiable feature selection and reconstruction[C]//Proceedings of the 36th International Conference on Machine Learning. San Diego, USA: JMLR, 2019: 444-453. [39] Emary E, Zawbaa H M, Hassanien A E.Binary grey wolf optimization approaches for feature selection[J]. Neurocomputing, 2016, 172: 371-381. [40] Nadimi-Shahraki M H, Taghian S, Mirjalili S. An improved grey wolf optimizer for solving engineering problems[J/OL]. Expert Systems with Applications, 2021, 166. https://doi.org/10.1016/j.eswa.2020.113917. [41] Li L J, Xuan M L, Lin Q Z, et al. An evolutionary multitasking algorithm with multiple filtering for high-dimensional feature selection[J]. IEEE Transactions on Evolutionary Computation, 2023, 27(4): 802-816. [42] Chen K, Xue B, Zhang M J, et al. Evolutionary multitasking for feature selection in high-dimensional classification via particle swarm optimization[J]. IEEE Transactions on Evolutionary Computation, 2022, 26(3): 446-460.